from pydantic import BaseModel


class Ai_model(BaseModel):
    ModelName: str
    ModelFunction: str
    InputType: str
    OutputType: str
    ModelAPIAddress: str

    class Config:
        orm_mode = True
